
#ifndef __EDNN_INPUT_H__
#define __EDNN_INPUT_H__



#include "ednn.h"



struct ednn_layer_input 
{
    struct ednn_layer super;
    struct ednn_tensor *input;
    ednn_shape3d_t input_shape;
};
typedef struct ednn_layer_input ednn_layer_input_t;

struct ednn_layer *ednn_input_layer(ednn_shape3d_t shape);
#define Input(shape) ednn_input_layer(shape)

#endif /* __EDNN_INPUT_H__ */
